/*
  * Copyright (c) 2016 MediaTek Inc.  All rights reserved.
  *
  * This software is available to you under a choice of one of two
  * licenses.  You may choose to be licensed under the terms of the GNU
  * General Public License (GPL) Version 2, available from the file
  * COPYING in the main directory of this source tree, or the
  * BSD license below:
  *
  *     Redistribution and use in source and binary forms, with or
  *     without modification, are permitted provided that the following
  *     conditions are met:
  *
  *      - Redistributions of source code must retain the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer.
  *
  *      - Redistributions in binary form must reproduce the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer in the documentation and/or other materials
  *        provided with the distribution.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */

#ifndef __AP_H__
#define __AP_H__


/* ============================================================= */
/*      Common definition */
/* ============================================================= */
#define MBSS_VLAN_INFO_GET(                                             \
	__pAd, __VLAN_VID, __VLAN_Priority, __FromWhichBSSID)				\
{                                                                       \
	if ((__FromWhichBSSID < __pAd->ApCfg.BssidNum) &&					\
		(__FromWhichBSSID < HW_BEACON_MAX_NUM) &&						\
		(__pAd->ApCfg.MBSSID[__FromWhichBSSID].wdev.VLAN_VID != 0)) {	\
		__VLAN_VID = __pAd->ApCfg.MBSSID[__FromWhichBSSID].wdev.VLAN_VID;	\
		__VLAN_Priority = __pAd->ApCfg.MBSSID[__FromWhichBSSID].wdev.VLAN_Priority; \
	}                                                                   \
}

/* ============================================================= */
/*      Function Prototypes */
/* ============================================================= */

BOOLEAN APBridgeToWirelessSta(IN PRTMP_ADAPTER pAd,
			      IN PUCHAR pHeader,
			      IN UINT HdrLen,
			      IN PUCHAR pData, IN UINT DataLen, IN ULONG fromwdsidx);

INT ApAllowToSendPacket(IN RTMP_ADAPTER * pAd,
			IN struct wifi_dev *wdev, IN PNDIS_PACKET pPacket, OUT UCHAR * pWcid);

INT APSendPacket(RTMP_ADAPTER *pAd, PNDIS_PACKET pPacket);

NDIS_STATUS APInsertPsQueue(IN PRTMP_ADAPTER pAd,
			    IN PNDIS_PACKET pPacket,
			    IN MAC_TABLE_ENTRY * pMacEntry, IN UCHAR QueIdx);

NDIS_STATUS APHardTransmit(RTMP_ADAPTER *pAd, TX_BLK *pTxBlk, UCHAR QueIdx);

VOID APRxEAPOLFrameIndicate(IN PRTMP_ADAPTER pAd,
			    IN MAC_TABLE_ENTRY * pEntry,
			    IN RX_BLK * pRxBlk, IN UCHAR FromWhichBSSID);

VOID APHandleRxDataFrame(RTMP_ADAPTER *pAd, RX_BLK *pRxBlk);

VOID APRxErrorHandle(RTMP_ADAPTER *pAd, RX_BLK *pRxBlk);

INT APCheckRxError(RTMP_ADAPTER *pAd, RXINFO_STRUC *pRxInfo, RX_BLK *pRxBlk);

BOOLEAN APChkCls2Cls3Err(IN RTMP_ADAPTER *pAd, IN UCHAR Wcid, IN HEADER_802_11 * pHeader);

VOID RTMPDescriptorEndianChange(UCHAR *pData, ULONG DescriptorType);

VOID RTMPFrameEndianChange(IN RTMP_ADAPTER *pAd,
			   IN UCHAR *pData, IN ULONG Dir, IN BOOLEAN FromRxDoneInt);

/* ap_assoc.c */

VOID APAssocStateMachineInit(IN PRTMP_ADAPTER pAd,
			     IN STATE_MACHINE * S, OUT STATE_MACHINE_FUNC Trans[]);

VOID MbssKickOutStas(RTMP_ADAPTER *pAd, INT apidx, USHORT Reason);
VOID APMlmeKickOutSta(RTMP_ADAPTER *pAd, UCHAR *staAddr, UCHAR Wcid, USHORT Reason);

#ifdef DOT11W_PMF_SUPPORT
VOID APMlmeKickOutAllSta(RTMP_ADAPTER *pAd, UCHAR apidx, USHORT Reason);
#endif /* DOT11W_PMF_SUPPORT */

VOID APCls3errAction(RTMP_ADAPTER *pAd, ULONG wcid, HEADER_802_11 *hdr);

/* ap_auth.c */

void APAuthStateMachineInit(IN PRTMP_ADAPTER pAd,
			    IN STATE_MACHINE * Sm, OUT STATE_MACHINE_FUNC Trans[]);

VOID APCls2errAction(RTMP_ADAPTER *pAd, ULONG wcid, HEADER_802_11 *hdr);

/* ap_connect.c */

#ifdef CONFIG_AP_SUPPORT
BOOLEAN BeaconTransmitRequired(RTMP_ADAPTER *pAd, INT apidx, MULTISSID_STRUCT *mbss);
#endif /* CONFIG_AP_SUPPORT */

VOID APMakeBssBeacon(RTMP_ADAPTER *pAd, INT apidx);
VOID APUpdateBeaconFrame(RTMP_ADAPTER *pAd, INT apidx);
VOID APMakeAllBssBeacon(RTMP_ADAPTER *pAd);
VOID APUpdateAllBeaconFrame(RTMP_ADAPTER *pAd);
void updateAllBeacon(RTMP_ADAPTER *pAd, INT apidx, ULONG FrameLen);

/* ap_sync.c */
VOID APSyncStateMachineInit(IN PRTMP_ADAPTER pAd,
			    IN STATE_MACHINE * Sm, OUT STATE_MACHINE_FUNC Trans[]);

VOID APScanTimeout(IN PVOID SystemSpecific1,
		   IN PVOID FunctionContext, IN PVOID SystemSpecific2, IN PVOID SystemSpecific3);

VOID ApSiteSurvey(IN PRTMP_ADAPTER pAd,
		  IN PNDIS_802_11_SSID pSsid, IN UCHAR ScanType, IN BOOLEAN ChannelSel);

VOID SupportRate(IN PUCHAR SupRate,
		 IN UCHAR SupRateLen,
		 IN PUCHAR ExtRate,
		 IN UCHAR ExtRateLen,
		 OUT PUCHAR * Rates, OUT PUCHAR RatesLen, OUT PUCHAR pMaxSupportRate);

BOOLEAN ApScanRunning(RTMP_ADAPTER *pAd);

#ifdef AP_PARTIAL_SCAN_SUPPORT
UCHAR FindPartialScanChannel(IN PRTMP_ADAPTER pAd);
#endif /* AP_PARTIAL_SCAN_SUPPORT */

VOID APUpdateOperationMode(RTMP_ADAPTER *pAd);

VOID APOverlappingBSSScan(RTMP_ADAPTER *pAd);

INT GetBssCoexEffectedChRange(IN RTMP_ADAPTER *pAd, IN BSS_COEX_CH_RANGE * pCoexChRange);

/* ap_mlme.c */
VOID APDetectOverlappingExec(IN PVOID SystemSpecific1,
			     IN PVOID FunctionContext,
			     IN PVOID SystemSpecific2, IN PVOID SystemSpecific3);

VOID APMlmePeriodicExec(RTMP_ADAPTER *pAd);

BOOLEAN APMsgTypeSubst(IN PRTMP_ADAPTER pAd,
		       IN PFRAME_802_11 pFrame, OUT INT *Machine, OUT INT *MsgType);

VOID APQuickResponeForRateUpExec(IN PVOID SystemSpecific1,
				 IN PVOID FunctionContext,
				 IN PVOID SystemSpecific2, IN PVOID SystemSpecific3);

VOID APAsicEvaluateRxAnt(RTMP_ADAPTER *pAd);
VOID APAsicRxAntEvalTimeout(RTMP_ADAPTER *pAd);

/* ap.c */
UCHAR get_apidx_by_addr(RTMP_ADAPTER *pAd, UCHAR *addr);

NDIS_STATUS APInitialize(RTMP_ADAPTER *pAd);
VOID APShutdown(RTMP_ADAPTER *pAd);
VOID APStartUp(RTMP_ADAPTER *pAd);
VOID APStop(RTMP_ADAPTER *pAd);

VOID APCleanupPsQueue(RTMP_ADAPTER *pAd, QUEUE_HEADER *pQueue);

VOID MacTableMaintenance(RTMP_ADAPTER *pAd);

UINT32 MacTableAssocStaNumGet(RTMP_ADAPTER *pAd);

MAC_TABLE_ENTRY *APSsPsInquiry(IN PRTMP_ADAPTER pAd,
			       IN PUCHAR pAddr,
			       OUT SST * Sst,
			       OUT USHORT *Aid, OUT UCHAR *PsMode, OUT UCHAR *Rate);

#ifdef SYSTEM_LOG_SUPPORT
VOID ApLogEvent(IN PRTMP_ADAPTER pAd, IN PUCHAR pAddr, IN USHORT Event);
#else
#define ApLogEvent(_pAd, _pAddr, _Event)
#endif /* SYSTEM_LOG_SUPPORT */

VOID APUpdateCapabilityAndErpIe(RTMP_ADAPTER *pAd);

BOOLEAN ApCheckAccessControlList(RTMP_ADAPTER *pAd, UCHAR *addr, UCHAR apidx);
VOID ApUpdateAccessControlList(RTMP_ADAPTER *pAd, UCHAR apidx);

BOOLEAN PeerAssocReqCmmSanity(IN PRTMP_ADAPTER pAd,
			      IN BOOLEAN isRessoc,
			      IN VOID *Msg, IN INT MsgLen, IN IE_LISTS * ie_lists);

BOOLEAN PeerDisassocReqSanity(IN PRTMP_ADAPTER pAd,
			      IN VOID *Msg,
			      IN ULONG MsgLen,
			      OUT PUCHAR pAddr2, OUT UINT16 * SeqNum, OUT USHORT *Reason);

BOOLEAN PeerDeauthReqSanity(IN PRTMP_ADAPTER pAd,
			    IN VOID *Msg,
			    IN ULONG MsgLen,
			    OUT PUCHAR pAddr2, OUT UINT16 * SeqNum, OUT USHORT *Reason);

BOOLEAN APPeerAuthSanity(IN PRTMP_ADAPTER pAd,
			 IN VOID *Msg,
			 IN ULONG MsgLen,
			 OUT PUCHAR pAddr1,
			 OUT PUCHAR pAddr2,
			 OUT USHORT *Alg,
			 OUT USHORT *Seq, OUT USHORT *Status, OUT CHAR * ChlgText
	);

#ifdef DOT1X_SUPPORT
INT Set_OwnIPAddr_Proc(RTMP_ADAPTER *pAd, PSTRING arg);
INT Set_EAPIfName_Proc(RTMP_ADAPTER *pAd, PSTRING arg);
INT Set_PreAuthIfName_Proc(RTMP_ADAPTER *pAd, PSTRING arg);

/* Define in ap.c */
BOOLEAN DOT1X_InternalCmdAction(IN RTMP_ADAPTER *pAd, IN MAC_TABLE_ENTRY *pEntry, IN UINT8 cmd);

BOOLEAN DOT1X_EapTriggerAction(RTMP_ADAPTER *pAd, MAC_TABLE_ENTRY *pEntry);
#endif /* DOT1X_SUPPORT */

VOID AP_E2PROM_IOCTL_PostCtrl(RTMP_IOCTL_INPUT_STRUCT *wrq, PSTRING msg);

VOID IAPP_L2_UpdatePostCtrl(RTMP_ADAPTER *pAd, UINT8 *mac, INT bssid);

#endif /* __AP_H__ */
